import os
from PyPDF2 import PdfMerger
import argparse

def merge_pdfs(input_files, output_path):
    """
    合并多个PDF文件
    
    Args:
        input_files (list): 输入PDF文件路径列表
        output_path (str): 输出PDF文件路径
    """
    # 创建PDF合并器
    merger = PdfMerger()
    
    # 添加所有PDF文件
    for pdf in input_files:
        merger.append(pdf)
    
    # 保存合并后的文件
    merger.write(output_path)
    merger.close()
    
    print(f'已合并PDF文件: {output_path}')

def main():
    parser = argparse.ArgumentParser(description='PDF合并工具')
    parser.add_argument('input_files', nargs='+', help='输入PDF文件路径（可以指定多个）')
    parser.add_argument('-o', '--output', default='merged.pdf', help='输出PDF文件路径（默认：merged.pdf）')
    
    args = parser.parse_args()
    
    # 确保输出目录存在
    output_dir = os.path.dirname(args.output)
    if output_dir:
        os.makedirs(output_dir, exist_ok=True)
    
    merge_pdfs(args.input_files, args.output)

if __name__ == '__main__':
    main() 